import os.path
from datetime import datetime

import webdriver_manager.drivers.chrome
from requests import options
from selenium import webdriver
from selenium.webdriver.ie.service import Service
from webdriver_manager.chrome import ChromeDriverManager


#创建一个浏览器的对象因为要对多个页面进行测试不可能每个页面都创建一个浏览器的对象
class Driver:
    #定义构造函数
    #声明变量
    driver = ""
    def __init__(self):
        options = webdriver.ChromeOptions()
        self.driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()),options=options)
        self.driver.maximize_window()
    #创建屏幕截图方面
    def getscreenshot(self):
        #需要屏幕截图的保存到images文件中
        #保存的文件名为2024—10—28—时间.png
        diam = datetime.now().strftime("%Y-%m-%d")
        #判断保存图片的日期文件夹是否存在
        if not os.path.exists("../images/"+diam):
            #不存在就创建一个日期分类的文件夹
            os.mkdir("../images/"+diam)
        #设置图片的名称格式
        fliename = datetime.now().strftime("%Y-%m-%d-%H%M%S")+".png"
        self.driver.save_screenshot("../images/"+diam+"/"+fliename)

forum = Driver()